# -*- coding: utf-8 -*-
'''
Author:     郭国辉
CreateTime:     2018年2月23日    下午5:09:01
'''
import random
import pygame
import file_utils

#游戏屏幕常量
SCREEN_RECT = pygame.Rect(0, 0, 480, 700)
# 刷新帧率
FRAMES_PER_SECOND = 60
#创建敌机定时器常量
ENEMY_EVENT = pygame.USEREVENT
#英雄移动速度
HERO_SPEED = 5
#英雄发射子弹事件
HERO_FIRE_EVENT = pygame.USEREVENT + 1
#子弹移动速度
BULLET_SPEED = 5

        
class GameSprite(pygame.sprite.Sprite):
    '''飞机精灵'''

    def __init__(self, img_name, speed=1):
        #调用父类init
        super().__init__()

        #定义对象的属性
        self.image = pygame.image.load(img_name)
        self.rect = self.image.get_rect()
        self.speed = speed

    def update(self):
        #垂直方向上移动
        self.rect.y += self.speed


class BackgroundSprite(GameSprite):
    '''游戏背景精灵'''

    def update(self):
        #垂直方向上移动
        super().update()

        #判断是否移出屏幕，如果移出屏幕将图像设置到屏幕上方
        if self.rect.y >= SCREEN_RECT.height:
            self.rect.y = -SCREEN_RECT.height


class EnemySprite(GameSprite):
    '''敌机精灵'''

    def __init__(self, img_name):
        super().__init__(img_name)
        self.speed = random.randint(1, 3)

        self.rect.y = -self.rect.height

        max_x = SCREEN_RECT.width - self.rect.width
        self.rect.x = random.randint(0, max_x)

    def update(self):
        super().update()
        #判断是否移出屏幕，如果移出屏幕从精灵组中删除该精灵
        if self.rect.y >= SCREEN_RECT.height:
            self.kill()
    

class HeroSprite(GameSprite):
    '''英雄精灵'''

    def __init__(self, img_name, speed=0):
        super().__init__(img_name, speed)

        self.rect.centerx = SCREEN_RECT.centerx
        self.rect.bottom = SCREEN_RECT.bottom - 120

        #创建子弹精灵组
        self.bullet_group = pygame.sprite.Group()
        

    def update(self):
        #英雄在水平方向移动
        self.rect.x += self.speed

#         right_x=SCREEN_RECT.width-self.rect.width

        if self.rect.x < 0:
            self.rect.x = 0
#         elif self.rect.x>right_x:
#             self.rect.x=right_x
        elif self.rect.right > SCREEN_RECT.right:
            self.rect.right = SCREEN_RECT.right

    def fire(self):
        for i in (0,1,2):
            self.__create_bullet(i*20);
            
    def __create_bullet(self,distance):
        #创建子弹精灵
        bullet_sprite = BulletSprite("%s\\images\\bullet_1.png" % file_utils.get_root())
        #设置子弹精灵的位置
        bullet_sprite.rect.bottom = self.rect.y - distance
        bullet_sprite.rect.centerx = self.rect.centerx
        #子弹精灵添加到精灵组
        self.bullet_group.add(bullet_sprite)
        
    def __del__(self):
        self.kill();

class BulletSprite(GameSprite):
    '''子弹精灵'''

    def __init__(self, img_name, speed=-BULLET_SPEED):
        super().__init__(img_name, speed)

    def update(self):
        super().update()
        #判断是否移出屏幕，如果移出屏幕从精灵组中删除该精灵
        if self.rect.bottom <= 0:
            self.kill()